%run train-cGAN
# %run unit_test

self.device is cuda:0
64
Epoch: 0, Itration: 0
loss_G_GAN = 4.886846542358398, loss_G_L1 = 0.39250922203063965 * 5
loss_D = 0.7143234014511108

Epoch: 76, Itration: 1000
loss_G_GAN = 9.659367561340332, loss_G_L1 = 0.15902411937713623 * 5
loss_D = 0.006910865660756826

Epoch: 153, Itration: 2000
loss_G_GAN = 8.78011417388916, loss_G_L1 = 0.17440780997276306 * 5
loss_D = 0.0005042690900154412

save model to ./checkpoints/pix2food.pkl at iteration = 3000
Epoch: 230, Itration: 3000
loss_G_GAN = 4.836752891540527, loss_G_L1 = 0.17835037410259247 * 5
loss_D = 0.18689367175102234

Finally save model to ./checkpoints/pix2food.pkl



 # --- Validation Set --- # 

# %run train-cGAN

# %run train